import os, sys
from BioModels import BioModels

class TestBioModels(object):

    def __init__(self):
        self.b = BioModels.BioModels('proxy.nbn.sun.ac.za:3128') 

    def FirstBatch(self):
        print self.b.helloBioModels()
        print self.b.getAllCuratedModelsId()
        print self.b.getAllModelsId()
        print self.b.getAllNonCuratedModelsId()
        print self.b.getModelById('BIOMD0000000091')

    def SecondBatch(self):
        print self.b.getModelSBMLById('BIOMD0000000091')
        print self.b.getSimpleModelById('BIOMD0000000091')
        #getModelsIdByChEBI(bio_id)
        #getModelsIdByChEBIId(bio_id)
        print self.b.getModelsIdByGO('GO0009408')
        print self.b.getModelsIdByName("Edelstein1996_EPSP_AChEvent")


    def ThirdBatch(self):        
        print self.b.getModelsIdByPerson('Gillespie')
        print self.b.getModelsIdByPublication('10.1098/rsif.2006.0148')
        print self.b.getModelsIdByTaxonomy('Saccharomyces cerevisiae')
    #    print self.b.getModelsIdByTaxonomyId()

    def FourthBatch(self):
        pass
    #    print self.b.getModelsIdByUniprot()

    #    print self.b.getModelsIdByUniprotId(bio_id)
    #    print self.b.getModelsIdByUniprotIds(bio_id)

    def FifthBatch(self):
    #    print self.b.getSimpleModelsByChEBIIds(bio_id)
    #    print self.b.getSimpleModelsByUniprotIds(bio_id)
    #    print self.b.getSimpleModelsRelatedWithChEBI()
        print self.b.getSimpleModelsByReactomeIds('REACT_152')
        
if __name__ == '__main__':
    t= TestBioModels()
    t.FirstBatch()
    t.SecondBatch()
    t.ThirdBatch()
    t.FourthBatch()
    t.FifthBatch()
    
